import xlsxwriter

# 创建一个新的 Excel 工作簿
workbook = xlsxwriter.Workbook('example_with_dropdown.xlsx')
worksheet = workbook.add_worksheet()

# 添加一些示例数据
worksheet.write('A1', 'ID')
worksheet.write('B1', '名称')
worksheet.write('C1', '选择（是/否）')

data = [
    [1, "Alice"],
    [2, "Bob"],
    [3, "Charlie"]
]

row = 1
for item in data:
    worksheet.write(row, 0, item[0])
    worksheet.write(row, 1, item[1])
    row += 1

# 设置数据验证为下拉选择 是/否
worksheet.data_validation('C2:C100', {
    'validate': 'list',
    'source': ['是', '否'],
    'input_message': '选择 是 或 否',
    'error_message': '只能选择 是 或 否',
})

# 关闭并保存工作簿
workbook.close()

print("文件已保存为 example_with_dropdown.xlsx，其中C列为是/否选择")
